import torch

# i = torch.LongTensor([[0,1,1]
#                       [2,0,2]
#                       ])
#
# v = torch.FloatTensor([3,4,5])
# sv = torch.sparse.FloatTensor(i,v,torch.size(2,3)).toDense()
# print(sv)
# 定义稀疏矩阵有值的(row, column)，例如第一个值在(0, 2)，第一/二列的第一栏
i = torch.LongTensor([[1, 1, 1,1],
                      [0, 0, 2,1]])
# 稀疏矩阵的值
v = torch.FloatTensor([3, 4, 5, 6])

# 定义稀疏矩阵的尺寸(2, 3)，并转为正常的矩阵
sv = torch.sparse.FloatTensor(i, v, torch.Size([2,3])).to_dense()
print(sv)

print(torch.cuda.is_available())